import os
from torch import distributed as dist


def setup_env():
    dist.init_process_group("gloo")


def cleanup_env():
    dist.destroy_process_group()
